package leetCode;

import java.util.Arrays;
import java.util.Scanner;

/*

给定一个由 n 个节点组成的网络，用 n x n 个邻接矩阵 graph 表示。在节点网络中，只有当 graph[i][j] = 1 时，节点 i 能够直接连接到另一个节点 j。

一些节点 initial 最初被恶意软件感染。只要两个节点直接连接，且其中至少一个节点受到恶意软件的感染，那么两个节点都将被恶意软件感染。这种恶意软件的传播将继续，直到没有更多的节点可以被这种方式感染。

假设 M(initial) 是在恶意软件停止传播之后，整个网络中感染恶意软件的最终节点数。

我们可以从 initial 中删除一个节点，并完全移除该节点以及从该节点到任何其他节点的任何连接。

请返回移除后能够使 M(initial) 最小化的节点。如果有多个节点满足条件，返回索引 最小的节点 。

 

示例 1：

输入：graph = [[1,1,0],[1,1,0],[0,0,1]], initial = [0,1]
输出：0
示例 2：

输入：graph = [[1,1,0],[1,1,1],[0,1,1]], initial = [0,1]
输出：1
示例 3：

输入：graph = [[1,1,0,0],[1,1,1,0],[0,1,1,1],[0,0,1,1]], initial = [0,1]
输出：1
 

提示：

n == graph.length
n == graph[i].length
2 <= n <= 300
graph[i][j] 是 0 或 1.
graph[i][j] == graph[j][i]
graph[i][i] == 1
1 <= initial.length < n
0 <= initial[i] <= n - 1
 initial 中每个整数都不同


5
1 1 0 0 0
1 1 1 0 0
0 1 1 1 1
0 0 1 1 0
0 0 1 0 1
2
1 2

9
1 0 0 0 0 0 0 0 0
0 1 0 0 0 0 0 0 0
0 0 1 0 1 0 1 0 0
0 0 0 1 0 0 0 0 0
0 0 1 0 1 0 0 0 0
0 0 0 0 0 1 0 0 0
0 0 1 0 0 0 1 0 0
0 0 0 0 0 0 0 1 0
0 0 0 0 0 0 0 0 1
3
6 0 4

3
1 1 0
1 1 1
0 1 1
2
0 1

4
1 1 1 0
1 1 1 0
1 1 1 0
0 0 0 1
3
0 1 3

9
1 0 0 0 0 0 0 0 1
0 1 0 1 0 0 0 0 0
0 0 1 1 0 1 0 0 0
0 1 1 1 1 0 1 0 0
0 0 0 1 1 1 0 0 0
0 0 1 0 1 1 0 0 0
0 0 0 1 0 0 1 1 0
0 0 0 0 0 0 1 1 1
1 0 0 0 0 0 0 0 1
2
3 7


5
1 1 1 0 0
1 1 0 0 1
1 0 1 1 1
0 0 1 1 0
0 1 1 0 1
2
0 3

5
1 1 1 0 0
1 1 0 1 0
1 0 1 1 0
0 1 1 1 1
0 0 0 1 1
2
0 4

4
1 1 0 1
1 1 1 1
0 1 1 0
1 1 0 1
2
0 2


 */

public class Solution928 {
    
    public int minMalwareSpread(int[][] graph, int[] initial) {
        int n = graph.length;
        Arrays.sort(initial);
        boolean[] infected = new boolean[n];
        for (int i : initial) {
            infected[i] = true;
        }
        int maxSave = 0;
        int maxSaveIndex = initial[0];
        for (int i : initial) {
            boolean[] visited = new boolean[n];
            visited[i] = true;
            int save = 1;
            for (int j = 0; j < n; j++) {
                if (i != j && graph[i][j] == 1) {
                    int temp = dfs(j, graph, visited, infected);
                    if (temp >= 0) {
                        save += temp;
                    }
                }
            }
            if (save > maxSave) {
                maxSave = save;
                maxSaveIndex = i;
            }
            System.out.println(save);
        }
        
        return maxSaveIndex;
    }
        

    private int dfs(int i, int[][] graph, boolean[] visited, boolean[] infected) {
        if (visited[i]) {
            return 0;
        } else {
            visited[i] = true;
        }
        if (infected[i]) {
            visited[i] = false;
            return -1;
        }
        int n = graph.length;
        int save = 1;
        for (int j = 0; j < n; j++) {
            if (i != j && graph[i][j] == 1) {
                int temp = dfs(j, graph, visited, infected);
                if (temp >= 0) {
                    save += temp;
                } else {
                    visited[i] = false;
                    return -1;
                }
            }
        }
        return save;
    }


    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int n = scan.nextInt();
        int[][] graph = new int[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                graph[i][j] = scan.nextInt();
            }
        }
        int m = scan.nextInt();
        int[] initial = new int[m];
        for (int i = 0; i < m; i++) {
            initial[i] = scan.nextInt();
        }
        Solution928 s = new Solution928();
        System.out.println(s.minMalwareSpread(graph, initial));

    }

}